﻿#include "Heap.h"

//初始化堆
void HeapInit(Hp* php)
{
	assert(php);
	php->arr = NULL;
	php->size = php->capacity = 0;
}

// 堆的销毁
void HeapDestory(Hp* php)
{
	assert(php);
	if (php->arr)
		free(php->arr);
	php->arr = NULL;
	php->size = php->capacity = 0;
}

void Swap(HPDataType* x, HPDataType* y)
{
	HPDataType tmp = *x;
	*x = *y;
	*y = tmp;
}
//向上调整
void AdjustUp(HPDataType* arr, int child)
{
	int parent = (child - 1) / 2;
	//建小堆
	while (child > 0)
	{
		//<小堆
		//>大堆
		if (arr[child] < arr[parent] )
		{
			Swap(&arr[child], &arr[parent]);
			child = parent;
			parent = (child - 1) / 2;
		}
		else
		{
			break;
		}
	}
}

//向下调整
void AdjustDown(HPDataType* arr, int parent, int n)
{
	HPDataType child = parent * 2 + 1;
	
	while (child < n)
	{
		//找最小的孩子--小堆
		//找最大的孩子--大堆
		
		if (arr[child] > arr[child + 1] && child+1<n)
		{
			child++;
		}
		//<小堆
		//>大堆
		if (arr[child] < arr[parent])
		{
			Swap(&arr[child], &arr[parent]);
			parent = child;
			child = parent * 2 + 1;
		}
		else
		{
			break;
		}
	}
}

// 堆的插入
void HeapPush(Hp* php, HPDataType x)
{
	assert(php);
	//检查是否需要增容
	if (php->size == php->capacity)
	{
		int newcapacity = php->capacity == 0 ? 4 : php->capacity * 2;
		HPDataType* tmp = (HPDataType*)realloc(php->arr, newcapacity * sizeof(HPDataType));
		if (tmp == NULL)
		{
			perror("realloc fail");
			exit(-1);
		}
		php->arr = tmp;
		php->capacity = newcapacity;
	}
	php->arr[php->size] = x;
	//尾部插入.向上调整
	AdjustUp(php->arr, php->size);
	++php->size;
}

// 堆的判空
bool HeapEmpty(Hp* php)
{
	assert(php);
	return php->size == 0;
}
// 堆的删除
void HeapPop(Hp* php)
{
	assert(!HeapEmpty(php));
	//交换第一个和最后一个元素
	Swap(&php->arr[0], &php->arr[php->size - 1]);
	php->size--;
	//向下调整
	AdjustDown(php->arr, 0,php->size);
}


// 取堆顶的数据
HPDataType HeapTop(Hp* php)
{
	assert(!HeapEmpty(php));
	return php->arr[0];
}
// 堆的数据个数
int HeapSize(Hp* php)
{
	assert(php);
	return php->size;
}
